import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.util.Arrays;

public class ImageSplitter {

    public static void main(String[] args) {
        // 设置要分割的目录
        File sourceDir = new File("C:/Users/18756/Desktop/photoselect/avatars"); // 替换为您的图片目录
        String[] extensions = new String[]{"jpg", "jpeg", "png", "gif"}; // 这里可以替换为需要的文件扩展名
        
        // 统计所有图像文件
        File[] imageFiles = sourceDir.listFiles((dir, name) -> {
            String lowerCaseName = name.toLowerCase();
            return Arrays.stream(extensions).anyMatch(lowerCaseName::endsWith);
        });

        if (imageFiles == null || imageFiles.length == 0) {
            System.out.println("没有找到图片文件.");
            return;
        }

        // 计算每份图像数量
        int totalFiles = imageFiles.length;
        int filesPerGroup = totalFiles / 20 + (totalFiles % 20 == 0 ? 0 : 1);

        // 创建20个子目录并移动文件
        for (int i = 0; i < 20; i++) {
            File subDir = new File(sourceDir, "group" + (i + 1));
            subDir.mkdir();

            for (int j = 0; j < filesPerGroup && (i * filesPerGroup + j) < totalFiles; j++) {
                File imageFile = imageFiles[i * filesPerGroup + j];
                try {
                    // 移动文件到子目录
                    Path targetPath = subDir.toPath().resolve(imageFile.getName());
                    Files.move(imageFile.toPath(), targetPath, StandardCopyOption.REPLACE_EXISTING);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        System.out.println("分割完成!");
    }
}